# 栅格化图像生成脚本

import os
import torch
from PIL import Image, ImageDraw, ImageFont

def generate_style_images(svg_metadata, output_image_dir, image_size=128):
    """生成风格图像（简化示例）"""
    os.makedirs(output_image_dir, exist_ok=True)
    
    # 示例：为每个字体生成一个样式图像
    for font_name in set(md['font'] for md in svg_metadata):
        # 创建空白图像
        img = Image.new('RGB', (image_size, image_size), (255, 255, 255))
        draw = ImageDraw.Draw(img)
        
        # 绘制字体名称
        try:
            font = ImageFont.truetype(f"data/raw_fonts/{font_name}.ttf", 32)
            draw.text((10, 50), font_name[:3], fill=(0, 0, 0), font=font)
        except:
            draw.text((10, 50), font_name[:3], fill=(0, 0, 0))
        
        # 保存图像张量
        tensor = torch.tensor(img).permute(2, 0, 1).float() / 255.0
        torch.save(tensor, os.path.join(output_image_dir, f"{font_name}_style.pt"))
